from langchain.tools import tool

class CustomTool:
    @tool("通用操作工具")
    def tool(self, action: str, **kwargs):
        """
        处理通用操作的工具
        参数:
            action: 操作类型
            **kwargs: 其他参数
        """
        if action == "format_response":
            # 格式化响应消息
            message = kwargs.get('message', '')
            data = kwargs.get('data', {})
            return f"{message}\n详细信息：{data}"
        
        elif action == "parse_input":
            # 解析用户输入
            user_input = kwargs.get('user_input', '')
            # 这里可以添加更复杂的输入解析逻辑
            return {
                "type": "unknown",
                "content": user_input
            }
        
        return f"未知操作: {action}"
